Using list comprehension to identify even or odd numbers in Python